/*
 *  Licensed to the Apache Software Foundation (ASF) under one or more
 *  contributor license agreements.  See the NOTICE file distributed with
 *  this work for additional information regarding copyright ownership.
 *  The ASF licenses this file to You under the Apache License, Version 2.0
 *  (the "License"); you may not use this file except in compliance with
 *  the License.  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package com.googlecode.httl.test;

import java.io.IOException;
import java.util.Date;
import java.util.Random;
import java.util.UUID;

import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;

import com.googlecode.httl.test.model.Book;
import com.googlecode.httl.test.model.User;

/**
 * DataFilter
 * 
 * @author Liang Fei (liangfei0201 AT gmail DOT com)
 */
public class DataFilter implements Filter {
    
    private final Random random = new Random();

    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException,
                                                                                     ServletException {
        int size = 100;
        String s = request.getParameter("size");
        if (s != null && s.length() > 0) {
            size = Integer.parseInt(s);
        }
        Book[] books = new Book[size];
        for (int i = 0; i < size; i ++) {
            books[i] = new Book(UUID.randomUUID().toString(), 
                                UUID.randomUUID().toString(), 
                                UUID.randomUUID().toString(), 
                                new Date(), 
                                random.nextInt(100) + 10, 
                                random.nextInt(60) + 30);
        }
        request.setAttribute("user", new User("liangfei", "admin"));
        request.setAttribute("books", books);
        chain.doFilter(request, response);
    }

    public void init(FilterConfig arg0) throws ServletException {
    }

    public void destroy() {
    }

}
